diff --git a/hikey960.mk b/hikey960.mk
index 3a979f8..d4e721f 100644
--- a/hikey960.mk
+++ b/hikey960.mk
@@ -48,10 +48,13 @@
   HIKEY_MOD_DIR := device/linaro/hikey-kernel/hikey960/$(TARGET_KERNEL_USE)
   HIKEY_MODS := $(wildcard $(HIKEY_MOD_DIR)/*.ko)
   SDCARDFS_KO := $(wildcard $(HIKEY_MOD_DIR)/sdcardfs*.ko)
+  CMA_HEAP_KO := $(wildcard $(HIKEY_MOD_DIR)/cma_heap.ko)
+  ION_CMA_HEAP_KO := $(wildcard $(HIKEY_MOD_DIR)/ion_cma_heap*.ko)
   ifneq ($(HIKEY_MODS),)
     BOARD_VENDOR_KERNEL_MODULES += $(HIKEY_MODS)
     BOARD_VENDOR_RAMDISK_KERNEL_MODULES += \
-	$(HIKEY_MOD_DIR)/ion_cma_heap.ko \
-	$(SDCARDFS_KO)
+        $(CMA_HEAP_KO) \
+        $(ION_CMA_HEAP_KO) \
+        $(SDCARDFS_KO)
   endif
 endif